Pass keyword arguments passed to choose_pool to the pool constructor. This enables additional configuration of the pool.

Standard information about the request

This is a: bug fix

This change affects: pycbc_bruke_bank, inference. However, since choose_pool is never called with any keyword arguments, things are functionally the same.

This change changes: configuration

This change: has appropriate unit tests, follows style guidelines (See e.g. PEP8)

This change is backwards compatible

Motivation

This enables specifying e.g. initalizer and initargs when creating a pool. Both the default multiprocessing library and schwimmbad support these arguments.

This will allow me to improve how the multiprocessing pool is configured for nessai, which will be a subsequent PR.

Contents

I've added kwargs to choose_pool. I've also updated SinglePool to allow arbitrary keyword arguments, this is consistent with SerialPool from schwimmbad.

Links to any issues or associated PRs

N/A

Testing performed

The changes are covered by existing tests.

Additional notes

• The author of this pull request confirms they will adhere to the code of conduct